cd ../..

gpus=$1

python -m trainer.IntplContrastiveTrainer predict \
    --config configs/model/CRNN_contrastive.yaml \
    --config configs/dataset/inference.yaml \
    --config configs/trainer/default.yaml \
    --config configs/experiment/intplcontrastive.yaml \
    --model.exp_name desed+CL/unlabeled_od/stu-tea-intpl/mask_ratio0.4_mask_len4/ \
    --ckpt_path /home/shaonian/SED/SED/logs/RE/desed/+AudioSet/label_mapping/+bsz24/checkpoints/epoch194_metric1.4930.ckpt \
    --trainer.devices=${gpus} 